package com.example.shuiyin.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.example.shuiyin.entity.Task;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;
import org.apache.ibatis.annotations.Update;

import java.util.List;

/**
 * 任务Mapper接口
 * 替代过时的WatermarkTaskMapper
 */
@Mapper
public interface TaskMapper extends BaseMapper<Task> {
    
    /**
     * 根据任务ID查询任务
     * 
     * @param taskId 任务ID
     * @return 任务
     */
    @Select("SELECT * FROM tasks WHERE task_id = #{taskId}")
    Task selectByTaskId(@Param("taskId") String taskId);
    
    /**
     * 根据任务ID和用户ID查询任务
     * 
     * @param taskId 任务ID
     * @param userId 用户ID
     * @return 任务
     */
    @Select("SELECT * FROM tasks WHERE task_id = #{taskId} AND user_id = #{userId}")
    Task selectByTaskIdAndUserId(@Param("taskId") String taskId, @Param("userId") Long userId);
    
    /**
     * 更新任务状态
     * 
     * @param taskId 任务ID
     * @param status 状态
     * @param processedItems 已处理项目数
     * @return 影响行数
     */
    @Update("UPDATE tasks SET status = #{status}, processed_items = #{processedItems}, updated_at = NOW() WHERE task_id = #{taskId}")
    int updateTaskStatus(@Param("taskId") String taskId, @Param("status") String status, @Param("processedItems") Integer processedItems);
    
    /**
     * 查询待处理的任务
     * 
     * @param limit 限制数量
     * @return 任务列表
     */
    @Select("SELECT * FROM tasks WHERE status = 'PENDING' AND type LIKE 'WATERMARK%' ORDER BY created_at ASC LIMIT #{limit}")
    List<Task> selectPendingTasks(@Param("limit") int limit);
    
    /**
     * 统计用户一天内创建的任务数
     * 
     * @param userId 用户ID
     * @return 任务数
     */
    @Select("SELECT COUNT(*) FROM tasks WHERE user_id = #{userId} AND type LIKE 'WATERMARK%' AND created_at > DATE_SUB(NOW(), INTERVAL 1 DAY)")
    int countTasksInLastDay(@Param("userId") Long userId);
    
    /**
     * 根据类型查询用户任务
     * 
     * @param userId 用户ID
     * @param type 任务类型
     * @return 任务列表
     */
    @Select("SELECT * FROM tasks WHERE user_id = #{userId} AND type = #{type} ORDER BY created_at DESC")
    List<Task> selectByUserIdAndType(@Param("userId") Long userId, @Param("type") String type);
    
    /**
     * 查询用户任务数量
     * 
     * @param userId 用户ID
     * @return 任务数量
     */
    @Select("SELECT COUNT(*) FROM tasks WHERE user_id = #{userId}")
    int countByUserId(@Param("userId") Long userId);
} 